#!/usr/bin/env python
# -*- coding: utf-8 -*-
#
#       lyrics.py
#       
#       Copyright 2011 www.soycode.com
#       
#       This program is free software; you can redistribute it and/or modify
#       it under the terms of the GNU General Public License as published by
#       the Free Software Foundation; either version 3 of the License, or
#       (at your option) any later version.
#       
#       This program is distributed in the hope that it will be useful,
#       but WITHOUT ANY WARRANTY; without even the implied warranty of
#       MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#       GNU General Public License for more details.
#       
#       You should have received a copy of the GNU General Public License
#       along with this program; if not, write to the Free Software
#       Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
#       MA 02110-1301, USA.

"""
Generates lyrics by mimicking fed text, also tracking number of syllables.
www.soycode.com
"""

# standard
import random
# installed
# package


def MimicDict(filename):
  """Builds a dictionary of words, with each word mapping to words to follow it."""
  words = open(filename).read().split()
  WordDict = {}
  WordDict[''] = words[0]
  for i in range(len(words) - 1):
    CurrentWord = words[i]
    NextWord = words[i + 1]
    if CurrentWord in WordDict:
      WordDict[CurrentWord].append(NextWord)
    else:
      WordDict[CurrentWord] = [NextWord]
  return WordDict


def Mimic(dict, word):
  """Given mimic dict and start word, returns 200 weighted random words."""
  mimicry = ''
  for i in range(200):
    if word in dict:
      word = random.choice(dict[word])
    else:
      word = random.choice(dict[''])
    mimicry = mimicry + word + ' '
  return mimicry

